export EVAL_ERTACACHE=0
export SAVE_ERROR=0
export CACHE_ADJUST=0
export TIMESTEP_ALIGNMENT=0
export NUM_FRAMES=51
export SKIP_LIST="[3,5,6,8,9,10,12,13,14,16,17,18,20,21,22,24,25,27]" ### 18
export DATA_TEST=2
export PROMPT_PATH="your_folder/ERTACache/VBench_full_info.json"
export CKPT_DIR="your_model_folder/THUDM/CogVideoX-2b"
export PROJECT_DOLDER="your_folder/ERTACache"
export PYTHONPATH=$PROJECT_DOLDER:$PYTHONPATH
export CKPT_DIR="your_model_folder/hpcai-tech"
export TEXT_DIR="your_model_folder/DeepFloyd"
export VAE_FOLDER="your_model_folder/PixArt-alpha"



############################## Notice ##############################
###### you should install the videosys,and repalce the /videosys/schedulers/scheduling_rflow_open_sora.py with ours ERTACache/ERTACache4Opensora1.2/scheduling_rflow_open_sora.py


######### original
export EVAL_ERTACACHE=0
export OUTPUT_DIR="./sample/original"
export SAVE_ERROR=1
CUDA_VISIBLE_DEVICES=0 python3 ertacache_opensora1.2.py


######### time-align
export EVAL_ERTACACHE=1
export OUTPUT_DIR="./sample/ertacache_timealign"
export SAVE_ERROR=1
export TIMESTEP_ALIGNMENT=1
CUDA_VISIBLE_DEVICES=0 python3 ertacache_opensora1.2.py



# ######### time-align + adjust
export EVAL_ERTACACHE=1
export DATA_FOLDER_ORI="./sample/original/tensor"
export ADJUST_FOLDER="./sample/ertacache_timealign"
export DATA_FOLDER_CACHE="$ADJUST_FOLDER/ertacache_tensor"
python3 $PROJECT_DOLDER/adjust_utils/adjust_data_analysis.py
export OUTPUT_DIR="./sample/ertacache_timealign_adjust"
export SAVE_ERROR=0
export CACHE_ADJUST=1
export TIMESTEP_ALIGNMENT=1
CUDA_VISIBLE_DEVICES=0 python3 ertacache_opensora1.2.py
